# -*- coding: utf-8 -*-
# @Time    : 2021-05-07 21:06
# @Author  : xiayzuhen
# @File    : ATM机题目.py
# @Email   : 2861526584@qq.com

# 两个类： ATM ,用户
# atm 机 ：
# 属性：钱，名字，位置
# 方法： 存钱、取钱

# 用户：
# 属性：账号，密码，钱
# 方法：存款，取款，查询 ，以及存取款的ATM 机的金额增减
# 增加菜单功能

class ATM():
    def __init__(self,money,name,location):
        self.money = money
        self.name = name
        self.locatian = location

    def add_money(self,money):   # 存钱放入ATM机中
        print(f"存入金额：{money}")
        self.money += money

    def take_money(self,money):    # 取钱从ATM机中
        print(f"取出金额：{money}")
        self.money -= money

class User():
    def __init__(self,name,passwd,money):
        self.name = name
        self.passwd = passwd
        self.money = money

    def save_money(self,money,atmobj):
        print(f"存入金额：{money}")
        self.money += money
        atmobj.money += money

    def get_money(self,money,atmobj):
        print(f"取出金额：{money}")
        if self.money >= money and atmobj.money >= money:
            self.money -= money
            atmobj.money -= money
        else:
            print("账户或存款机余额不足")

    def show_money(self):
        print(f"当前余额账户为：{self.money}")

atm1 = ATM(10000, "工商银行", "nongda")
atm2 = ATM(20000, "建设银行", "gongda")
user1 = User("lilu", "123456", 50000)
user2 = User("xiayuzhen", "xyz", 50000)
# atm1.add_money(60000)
# print(atm1.money)

while True:
    print("""
    1、查询账户余额
    2、存款功能
    3、取款功能
    4、退出系统
    """)
    num = input("请选择你需要的功能：")
    if num == "1":
        print(f"您的账户余额为：{user1.money}")
        print("-"*20)

    if num == "2":
        Save_money = float(input(f"请输入您要存入的金额："))
        user1.save_money(Save_money,atm1)
        print("-"*20)

    if num == "3":
        Get_money = float(input(f"请输入您要取款的金额："))
        user1.get_money(Get_money,atm1)
        print("-"*20)

    if num == "4":
        print("退出系统")
        break










